package org.jsola.hr.service.impl;

import org.jsola.core.Page;
import org.jsola.hr.common.ExampleKit;
import org.jsola.hr.constant.HrConstants;
import org.jsola.hr.dao.IAttendShiftStaffScheduleDAO;
import org.jsola.hr.dto.AttendShiftStaffScheduleAddDTO;
import org.jsola.hr.dto.AttendShiftStaffScheduleUpdateDTO;
import org.jsola.hr.entity.AttendShiftStaffScheduleDO;
import org.jsola.hr.query.AttendShiftStaffScheduleQuery;
import org.jsola.hr.service.IAttendShiftStaffScheduleService;
import org.jsola.hr.vo.AttendShiftStaffScheduleListVO;
import org.jsola.hr.vo.AttendShiftStaffScheduleVO;
import org.jsola.user.core.TokenUser;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import tk.mybatis.mapper.entity.Example;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author zhr
 */
@Service("hrAttendShiftStaffScheduleServiceImpl")
public class AttendShiftStaffScheduleServiceImpl implements IAttendShiftStaffScheduleService {

    @Autowired
    private IAttendShiftStaffScheduleDAO attendShiftStaffScheduleDAO;

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public AttendShiftStaffScheduleVO save(AttendShiftStaffScheduleAddDTO attendShiftStaffScheduleAddDTO, TokenUser tokenUser) {
        // 转DO
        AttendShiftStaffScheduleDO attendShiftStaffScheduleDO = attendShiftStaffScheduleAddDTO.to(AttendShiftStaffScheduleDO.class);
        // 保存
        attendShiftStaffScheduleDO = save(attendShiftStaffScheduleDO, tokenUser.getSiteId(), tokenUser.getUserId());
        return attendShiftStaffScheduleDO.to(AttendShiftStaffScheduleVO.class);
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int update(AttendShiftStaffScheduleUpdateDTO attendShiftStaffScheduleUpdateDTO, TokenUser tokenUser) {
        //转DO
        AttendShiftStaffScheduleDO attendShiftStaffScheduleDO = attendShiftStaffScheduleUpdateDTO.to(AttendShiftStaffScheduleDO.class);
        //根据主键更新，只更新非null值
        return updateByIdSelective(attendShiftStaffScheduleDO, tokenUser.getSiteId(), tokenUser.getUserId());
    }

    @Override
    @Transactional(value = HrConstants.TX, rollbackFor = Exception.class)
    public int deleteByIds(TokenUser tokenUser, String... attendShiftStaffScheduleIds) {
        return logicalDeleteByIds(tokenUser.getSiteId(), tokenUser.getUserId(), (Object[]) attendShiftStaffScheduleIds);
    }


    @Override
    public AttendShiftStaffScheduleVO selectById(String attendShiftStaffScheduleId, String siteId) {
        AttendShiftStaffScheduleDO attendShiftStaffScheduleDO = selectDOById(attendShiftStaffScheduleId, siteId);
        if (attendShiftStaffScheduleDO == null) {
            return null;
        }
        return attendShiftStaffScheduleDO.to(AttendShiftStaffScheduleVO.class);
    }

    @Override
    public List<AttendShiftStaffScheduleListVO> select(AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery, String siteId) {
        List<AttendShiftStaffScheduleDO> attendShiftStaffScheduleDOList = selectDO(attendShiftStaffScheduleQuery, siteId);
        if(CollectionUtils.isEmpty(attendShiftStaffScheduleDOList)) {
            return attendShiftStaffScheduleDOList == null ? null : new ArrayList<>();
        }
        return attendShiftStaffScheduleDOList.stream()
                .map(attendShiftStaffScheduleDO -> attendShiftStaffScheduleDO.to(AttendShiftStaffScheduleListVO.class))
                .collect(Collectors.toList());
    }

    @Override
    public int selectCount(AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery, String siteId) {
        Example example = buildExample(attendShiftStaffScheduleQuery,siteId);
        return attendShiftStaffScheduleDAO.selectCountByExample(example);
    }

    @Override
    public Page<AttendShiftStaffScheduleListVO> selectPage(AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery, String siteId) {
        Example example = buildExample(attendShiftStaffScheduleQuery,siteId);
        Page<AttendShiftStaffScheduleDO> page = attendShiftStaffScheduleDAO.selectPageByExample(example,
                attendShiftStaffScheduleQuery.getPageNo(),
                attendShiftStaffScheduleQuery.getPageSize());

        return page.to(AttendShiftStaffScheduleListVO.class);
    }

    @Override
    public AttendShiftStaffScheduleDO selectDOById(String attendShiftStaffScheduleId, String siteId) {
        return listById(attendShiftStaffScheduleId, siteId);
    }

    @Override
    public List<AttendShiftStaffScheduleDO> selectDO(AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery, String siteId) {
        Example example = buildExample(attendShiftStaffScheduleQuery,siteId);
        return attendShiftStaffScheduleDAO.selectByExample(example);
    }

    @Override
    public int deleteByAttendPlanId(String attendPlanId, TokenUser tokenUser) {
        AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery = new AttendShiftStaffScheduleQuery();
        attendShiftStaffScheduleQuery.setAttendPlanId(attendPlanId);
        Example example = buildExample(attendShiftStaffScheduleQuery, tokenUser.getSiteId());
        return attendShiftStaffScheduleDAO.deleteByExample(example);
    }

    @Override
    public int updateAttendPlanId(String tmpAttendPlanId, String attendPlanId, TokenUser tokenUser) {
        AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery = new AttendShiftStaffScheduleQuery();
        attendShiftStaffScheduleQuery.setAttendPlanId(attendPlanId);
        Example example = buildExample(attendShiftStaffScheduleQuery, tokenUser.getSiteId());

        AttendShiftStaffScheduleDO record = new AttendShiftStaffScheduleDO();
        record.setAttendPlanId(attendPlanId);

        return attendShiftStaffScheduleDAO.updateByExampleSelective(record, example);
    }

    @Override
    public int logicDelByAttendPlanId(String attendPlanId, TokenUser tokenUser) {

        // 查询
        AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery = new AttendShiftStaffScheduleQuery();
        attendShiftStaffScheduleQuery.setAttendPlanId(attendPlanId);
        List<AttendShiftStaffScheduleDO> dataList = this.selectDO(attendShiftStaffScheduleQuery, tokenUser.getSiteId());
        if (dataList.size() <= 0) {
            return 0;
        }

        for (AttendShiftStaffScheduleDO attendShiftStaffSchedule : dataList) {

            AttendShiftStaffScheduleDO attendShiftStaffScheduleDO = new AttendShiftStaffScheduleDO();

            attendShiftStaffScheduleDO.setId(attendShiftStaffSchedule.getId());
            attendShiftStaffScheduleDO.setValid(false);
            attendShiftStaffScheduleDO.setModifiedUserId(tokenUser.getUserId());
            attendShiftStaffScheduleDO.setGmtModified(new Date());

            attendShiftStaffScheduleDAO.updateByPrimaryKeySelective(attendShiftStaffScheduleDO);
        }

        return 0;
    }

    /**
     * 根据查询参数，构建example

     * @param attendShiftStaffScheduleQuery 查询参数
     * @param siteId 所属站点id
     * @return example
     */
    private Example buildExample(AttendShiftStaffScheduleQuery attendShiftStaffScheduleQuery, String siteId) {
        Example example = new Example(AttendShiftStaffScheduleDO.class);
        example.and()
                .andEqualTo("valid", true)
                .andEqualTo("siteId", siteId);
        if (attendShiftStaffScheduleQuery.getAttendPlanId() != null) {
            example.and().andLike("attendPlanId", "%" + attendShiftStaffScheduleQuery.getAttendPlanId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getCompanyId() != null) {
            example.and().andLike("companyId", "%" + attendShiftStaffScheduleQuery.getCompanyId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getDeptInfoId() != null) {
            example.and().andLike("deptInfoId", "%" + attendShiftStaffScheduleQuery.getDeptInfoId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getEmpInfoId() != null) {
            example.and().andLike("empInfoId", "%" + attendShiftStaffScheduleQuery.getEmpInfoId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getYearly() != null) {
            example.and().andEqualTo("yearly", attendShiftStaffScheduleQuery.getYearly());
        }
        if (attendShiftStaffScheduleQuery.getMonthly() != null) {
            example.and().andEqualTo("monthly", attendShiftStaffScheduleQuery.getMonthly());
        }
        if (attendShiftStaffScheduleQuery.getDaily() != null) {
            example.and().andEqualTo("daily", attendShiftStaffScheduleQuery.getDaily());
        }
        if (attendShiftStaffScheduleQuery.getAttendShiftId() != null) {
            example.and().andLike("attendShiftId", "%" + attendShiftStaffScheduleQuery.getAttendShiftId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getAttendShiftName() != null) {
            example.and().andLike("attendShiftName", "%" + attendShiftStaffScheduleQuery.getAttendShiftName() + "%");
        }
        if (attendShiftStaffScheduleQuery.getType() != null) {
            example.and().andEqualTo("type", attendShiftStaffScheduleQuery.getType());
        }
        if (attendShiftStaffScheduleQuery.getGmtCreateStart() != null &&
                attendShiftStaffScheduleQuery.getGmtCreateEnd() != null) {
            example.and().andBetween("gmtCreate", attendShiftStaffScheduleQuery.getGmtCreateStart(),
                    attendShiftStaffScheduleQuery.getGmtCreateEnd());
        }
        if (attendShiftStaffScheduleQuery.getCreateUserId() != null) {
            example.and().andLike("createUserId", "%" + attendShiftStaffScheduleQuery.getCreateUserId() + "%");
        }
        if (attendShiftStaffScheduleQuery.getGmtModifiedStart() != null &&
                attendShiftStaffScheduleQuery.getGmtModifiedEnd() != null) {
            example.and().andBetween("gmtModified", attendShiftStaffScheduleQuery.getGmtModifiedStart(),
                    attendShiftStaffScheduleQuery.getGmtModifiedEnd());
        }
        if (attendShiftStaffScheduleQuery.getModifiedUserId() != null) {
            example.and().andLike("modifiedUserId", "%" + attendShiftStaffScheduleQuery.getModifiedUserId() + "%");
        }
        // 排序
        ExampleKit.setExampleOrder(example,attendShiftStaffScheduleQuery.getOrders());
        return example;
    }
}




